#!/bin/bash

# 程序名称
PROGRAM="tcp_server"
# 日志目录
LOG_DIR="logs"

# 创建日志目录
mkdir -p $LOG_DIR

# 启动程序
start() {
    if pgrep -x "$PROGRAM" > /dev/null; then
        echo "$PROGRAM 已经在运行中"
    else
        nohup ./$PROGRAM > /dev/null 2>&1 &
        echo "$PROGRAM 已启动"
    fi
}

# 停止程序
stop() {
    if pgrep -x "$PROGRAM" > /dev/null; then
        pkill -x "$PROGRAM"
        echo "$PROGRAM 已停止"
    else
        echo "$PROGRAM 未在运行"
    fi
}

# 查看状态
status() {
    if pgrep -x "$PROGRAM" > /dev/null; then
        echo "$PROGRAM 正在运行"
        ps aux | grep "$PROGRAM" | grep -v grep
    else
        echo "$PROGRAM 未在运行"
    fi
}

# 根据参数执行相应操作
case "$1" in
    start)
        start
        ;;
    stop)
        stop
        ;;
    restart)
        stop
        sleep 1
        start
        ;;
    status)
        status
        ;;
    *)
        echo "用法: $0 {start|stop|restart|status}"
        exit 1
        ;;
esac

exit 0 